'''
自定义代理

一个代理由二个部分组成：
    工具tool：代理可以使用的工具
    代理执行器：这决定了采取哪些行动
'''

# 导入包
from langchain.agents import Tool, AgentExecutor, BaseSingleActionAgent
from langchain import OpenAI, SerpAPIWrapper

import os
# # 设置全集OPEN_API_KEY值
# os.environ['OPENAI_API_KEY']="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"

# SerpApi是一个实时API，可以访问Google搜索结果。由于网络原因我们不方便演示
os.environ['SERPAPI_API_KEY']="xxx"


search = SerpAPIWrapper()
tools = [
    Tool(
        name="Search",
        func=search.run,
        description="useful for when you need to answer questions about current events",
        return_direct=True
    )
]
